tensor([[[0.2700, 0.2700],
         [0.2700, 0.2700]],

        [[0.3500, 0.3500],
         [0.3500, 0.3500]],

        [[0.3800, 0.3800],
         [0.3800, 0.3800]]], grad_fn=<RoundBackward1>)